package top.abrsm.util;

import org.apache.commons.lang3.StringUtils;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.client.HttpClient;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpDelete;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.conn.ClientConnectionManager;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.scheme.SchemeRegistry;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;

import javax.net.ssl.SSLContext;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.security.cert.X509Certificate;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class HttpUtils {

    /**
     * get
     *
     * @param host
     * @param path
     * @param method
     * @param headers
     * @param querys
     * @return
     * @throws Exception
     */
    public static HttpResponse doGet ( String host, String path, String method,
                                       Map<String, String> headers,
                                       Map<String, String> querys )
            throws Exception {
        HttpClient httpClient = wrapClient ( host );

        HttpGet request = new HttpGet ( buildUrl ( host, path, querys ) );
        for ( Map.Entry<String, String> e : headers.entrySet () ) {
            request.addHeader ( e.getKey (), e.getValue () );
        }

        return httpClient.execute ( request );
    }

    /**
     * post form
     *
     * @param host
     * @param path
     * @param method
     * @param headers
     * @param querys
     * @param bodys
     * @return
     * @throws Exception
     */
    public static HttpResponse doPost ( String host, String path, String method,
                                        Map<String, String> headers,
                                        Map<String, String> querys,
                                        Map<String, String> bodys )
            throws Exception {
        HttpClient httpClient = wrapClient ( host );

        HttpPost request = new HttpPost ( buildUrl ( host, path, querys ) );
        for ( Map.Entry<String, String> e : headers.entrySet () ) {
            request.addHeader ( e.getKey (), e.getValue () );
        }

        if ( bodys != null ) {
            List<NameValuePair> nameValuePairList = new ArrayList<NameValuePair> ();

            for ( String key : bodys.keySet () ) {
                nameValuePairList.add ( new BasicNameValuePair ( key, bodys.get ( key ) ) );
            }
            UrlEncodedFormEntity formEntity = new UrlEncodedFormEntity ( nameValuePairList, "utf-8" );
            formEntity.setContentType ( "application/x-www-form-urlencoded; charset=UTF-8" );
            request.setEntity ( formEntity );
        }

        return httpClient.execute ( request );
    }

    /**
     * Post String
     *
     * @param host
     * @param path
     * @param method
     * @param headers
     * @param querys
     * @param body
     * @return
     * @throws Exception
     */
    public static HttpResponse doPost ( String host, String path, String method,
                                        Map<String, String> headers,
                                        Map<String, String> querys,
                                        String body )
            throws Exception {
        HttpClient httpClient = wrapClient ( host );

        HttpPost request = new HttpPost ( buildUrl ( host, path, querys ) );
        for ( Map.Entry<String, String> e : headers.entrySet () ) {
            request.addHeader ( e.getKey (), e.getValue () );
        }

        if ( StringUtils.isNotBlank ( body ) ) {
            request.setEntity ( new StringEntity ( body, "utf-8" ) );
        }

        return httpClient.execute ( request );
    }

    /**
     * Post stream
     *
     * @param host
     * @param path
     * @param method
     * @param headers
     * @param querys
     * @param body
     * @return
     * @throws Exception
     */
    public static HttpResponse doPost ( String host, String path, String method,
                                        Map<String, String> headers,
                                        Map<String, String> querys,
                                        byte[] body )
            throws Exception {
        HttpClient httpClient = wrapClient ( host );

        HttpPost request = new HttpPost ( buildUrl ( host, path, querys ) );
        for ( Map.Entry<String, String> e : headers.entrySet () ) {
            request.addHeader ( e.getKey (), e.getValue () );
        }

        if ( body != null ) {
            request.setEntity ( new ByteArrayEntity ( body ) );
        }

        return httpClient.execute ( request );
    }

    /**
     * Put String
     *
     * @param host
     * @param path
     * @param method
     * @param headers
     * @param querys
     * @param body
     * @return
     * @throws Exception
     */
    public static HttpResponse doPut ( String host, String path, String method,
                                       Map<String, String> headers,
                                       Map<String, String> querys,
                                       String body )
            throws Exception {
        HttpClient httpClient = wrapClient ( host );

        HttpPut request = new HttpPut ( buildUrl ( host, path, querys ) );
        for ( Map.Entry<String, String> e : headers.entrySet () ) {
            request.addHeader ( e.getKey (), e.getValue () );
        }

        if ( StringUtils.isNotBlank ( body ) ) {
            request.setEntity ( new StringEntity ( body, "utf-8" ) );
        }

        return httpClient.execute ( request );
    }

    /**
     * Put stream
     *
     * @param host
     * @param path
     * @param method
     * @param headers
     * @param querys
     * @param body
     * @return
     * @throws Exception
     */
    public static HttpResponse doPut ( String host, String path, String method,
                                       Map<String, String> headers,
                                       Map<String, String> querys,
                                       byte[] body )
            throws Exception {
        HttpClient httpClient = wrapClient ( host );

        HttpPut request = new HttpPut ( buildUrl ( host, path, querys ) );
        for ( Map.Entry<String, String> e : headers.entrySet () ) {
            request.addHeader ( e.getKey (), e.getValue () );
        }

        if ( body != null ) {
            request.setEntity ( new ByteArrayEntity ( body ) );
        }

        return httpClient.execute ( request );
    }

    /**
     * Delete
     *
     * @param host
     * @param path
     * @param method
     * @param headers
     * @param querys
     * @return
     * @throws Exception
     */
    public static HttpResponse doDelete ( String host, String path, String method,
                                          Map<String, String> headers,
                                          Map<String, String> querys )
            throws Exception {
        HttpClient httpClient = wrapClient ( host );

        HttpDelete request = new HttpDelete ( buildUrl ( host, path, querys ) );
        for ( Map.Entry<String, String> e : headers.entrySet () ) {
            request.addHeader ( e.getKey (), e.getValue () );
        }

        return httpClient.execute ( request );
    }

    private static String buildUrl ( String host, String path, Map<String, String> querys ) throws UnsupportedEncodingException {
        StringBuilder sbUrl = new StringBuilder ();
        sbUrl.append ( host );
        if ( ! StringUtils.isBlank ( path ) ) {
            sbUrl.append ( path );
        }
        if ( null != querys ) {
            StringBuilder sbQuery = new StringBuilder ();
            for ( Map.Entry<String, String> query : querys.entrySet () ) {
                if ( 0 < sbQuery.length () ) {
                    sbQuery.append ( "&" );
                }
                if ( StringUtils.isBlank ( query.getKey () ) && ! StringUtils.isBlank ( query.getValue () ) ) {
                    sbQuery.append ( query.getValue () );
                }
                if ( ! StringUtils.isBlank ( query.getKey () ) ) {
                    sbQuery.append ( query.getKey () );
                    if ( ! StringUtils.isBlank ( query.getValue () ) ) {
                        sbQuery.append ( "=" );
                        sbQuery.append ( URLEncoder.encode ( query.getValue (), "utf-8" ) );
                    }
                }
            }
            if ( 0 < sbQuery.length () ) {
                sbUrl.append ( "?" ).append ( sbQuery );
            }
        }

        return sbUrl.toString ();
    }

    private static HttpClient wrapClient ( String host ) {
        HttpClient httpClient = new DefaultHttpClient ();
        if ( host.startsWith ( "https://" ) ) {
            sslClient ( httpClient );
        }

        return httpClient;
    }

    private static void sslClient ( HttpClient httpClient ) {
        try {
            SSLContext ctx = SSLContext.getInstance ( "TLS" );
            X509TrustManager tm = new X509TrustManager () {
                public X509Certificate[] getAcceptedIssuers () {
                    return null;
                }

                public void checkClientTrusted ( X509Certificate[] xcs, String str ) {

                }

                public void checkServerTrusted ( X509Certificate[] xcs, String str ) {

                }
            };
            ctx.init ( null, new TrustManager[]{tm}, null );
            SSLSocketFactory ssf = new SSLSocketFactory ( ctx );
            ssf.setHostnameVerifier ( SSLSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER );
            ClientConnectionManager ccm = httpClient.getConnectionManager ();
            SchemeRegistry registry = ccm.getSchemeRegistry ();
            registry.register ( new Scheme ( "https", 443, ssf ) );
        } catch ( KeyManagementException ex ) {
            throw new RuntimeException ( ex );
        } catch ( NoSuchAlgorithmException ex ) {
            throw new RuntimeException ( ex );
        }
    }
}
